FROM nvidia/cuda:12.2.2-base-ubuntu22.04

COPY . /app
WORKDIR /app
RUN apt-get update && \
    apt-get install -y python3 && \
    apt-get install -y python3-setuptools && \
    apt-get install -y python3-pip
RUN /usr/bin/pip3 install -r ./requirements.txt -i https://mirrors.aliyun.com/pypi/simple/  --trusted-host mirrors.aliyun.com
RUN pip3 install cn_clip
EXPOSE 8000
ENTRYPOINT python3 app/main.py